import onnx

if __name__ == '__main__':
    model = onnx.load_model("../logs/baseline_ljspeech/vits_encoder.onnx")
    for n in model.graph.node:
        print(n.op_type)